package com.bjpowernode.handler;

import cn.hutool.json.JSONUtil;
import cn.hutool.jwt.JWTUtil;
import com.bjpowernode.constant.Constant;
import com.bjpowernode.entity.TUser;
import com.bjpowernode.result.Result;
import jakarta.annotation.Resource;
import jakarta.servlet.ServletException;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.security.core.Authentication;
import org.springframework.security.web.authentication.AuthenticationSuccessHandler;
import org.springframework.stereotype.Component;

import java.io.IOException;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 * 应用认证成功处理器
 * 实现AuthenticationSuccessHandler接口，用于处理用户登录认证成功后的业务逻辑
 * 主要功能：生成JWT令牌、存储令牌到Redis、构建并返回登录成功响应
 */
@Component
public class AppAuthenticationSuccessHandler implements AuthenticationSuccessHandler {
    /**
     * Redis操作模板
     * 用于将生成的JWT令牌存储到Redis中，实现令牌的分布式管理
     */
    @Resource
    private RedisTemplate<String, Object> redisTemplate;

    /**
     * 认证成功后的处理方法
     * @param request HTTP请求对象
     * @param response HTTP响应对象
     * @param authentication 认证信息对象，包含已认证的用户信息
     * @throws IOException IO异常
     * @throws ServletException Servlet异常
     */
    @Override
    public void onAuthenticationSuccess(HttpServletRequest request, HttpServletResponse response, Authentication authentication) throws IOException, ServletException {
        // 设置响应内容类型为JSON格式，字符编码为UTF-8
        response.setContentType("application/json");
        response.setCharacterEncoding("utf-8");

        // 从认证信息中获取已认证的用户主体对象（TUser类型）
        TUser tUser = (TUser) authentication.getPrincipal();
        // 将用户对象序列化为JSON字符串，作为JWT令牌的载荷数据
        String userJSON = JSONUtil.toJsonStr(tUser);
        // 使用Hutool的JWT工具生成令牌
        // 参数1：载荷数据（包含用户信息）
        // 参数2：签名密钥（从系统常量中获取）
        String token = JWTUtil.createToken(Map.of("user", userJSON), Constant.SECRET.getBytes());

        // 将令牌存储到Redis
        // 键：用户ID拼接系统常量中的令牌前缀
        // 值：生成的JWT令牌
        // 过期时间：30分钟（与令牌过滤器中的刷新逻辑保持一致）
        redisTemplate.opsForValue().set(Constant.REDIS_TOKEN_KEY + tUser.getId(), token, 30L, TimeUnit.MINUTES);

        // 构建登录成功响应结果对象
        // 使用构建器模式设置状态码、消息和令牌信息
        Result result = Result.builder().code(200).msg("登录成功").info(token).build();

        // 将Result对象转换为JSON字符串
        String json = JSONUtil.toJsonStr(result);

        // 将JSON响应写入客户端
        response.getWriter().write(json);
    }
}
